import os
from openai import OpenAI
os.environ["DASHSCOPE_API_KEY"] = "sk-5c6689dccd074a739c78ef7d1d780148"

try:
    client = OpenAI(
        # 新加坡和北京地域的API Key不同。获取API Key：https://help.aliyun.com/zh/model-studio/get-api-key
        # 若没有配置环境变量，请用阿里云百炼API Key将下行替换为：api_key="sk-xxx",
        api_key=os.getenv("DASHSCOPE_API_KEY"),
        # 以下是北京地域base_url，如果使用新加坡地域的模型，需要将base_url替换为：https://dashscope-intl.aliyuncs.com/compatible-mode/v1
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    )

    completion = client.chat.completions.create(
        # 模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
        model="qwen-plus",
        messages=[
            {"role": "system", "content": '''
            
            
                        你需要帮我提取时间user中的content中所包含的时间：
                        
                        例子1："1分钟的成功率"，需要提取出来的是1分钟，只需要返回的是1，
                        
                        例子2："1个小时的成功率"，需要提取出来的是1个小时，转换为分钟为60分钟，只需要返回的是60，
                        
                        例子3："1天的成功率"，需要提取出来的是1天，转换为分钟为1440分钟，只需要返回的是1440，
                        
                        ...
                        
                        以此类推，你需要会时间的转换。
                        
                        如果是小时，则转换为分钟，如果是分钟，则还是分钟，
                        
                        最后只需要回复多少分钟即可，即只需要回复数字就行。
                        
                        
                        禁止回答其余的内容，
            
            
            '''},
            {"role": "user", "content": '''
            3小时的成功率
            '''},
        ],
    )
    print(completion.choices[0].message.content)
    # 如需查看完整响应，请取消下列注释
    # print(completion.model_dump_json())
except Exception as e:
    print(f"错误信息：{e}")
    print("请参考文档：https://help.aliyun.com/zh/model-studio/developer-reference/error-code")